package com.gh.rabbitmq.four;

import com.gh.rabbitmq.utils.RabbitmqUtils;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.ConfirmCallback;
import com.rabbitmq.client.ConfirmListener;

import java.util.UUID;
import java.util.concurrent.ConcurrentNavigableMap;
import java.util.concurrent.ConcurrentSkipListMap;

/**
 * @Author Eric
 * @Date 2021/7/17 23:00
 * @Version 1.0
 * 发布确认模式
 * 1.单个确认
 * 2.批量确认
 * 3.异步批量确认
 */
public class ConfirmMessage {

    //批量发消息的个数
    public static final int MESSAGE_COUNT = 1000;

    public static void main(String[] args) throws Exception {
        //单个确认
        //publishMessageIndividually();//发布1000个单独确认消息,耗时1700毫秒
        //批量确认
        //publishMessageBatch();//发布1000个批量确认消息,耗时356毫秒
        //异步确认
        publishMessageAsync();//发布1000个异步确认消息,耗时176毫秒
    }

    /**
     * 单个发布确认
     * @throws Exception
     */
    public static void publishMessageIndividually() throws Exception {
        Channel channel = RabbitmqUtils.getChannel();
        //队列的名称
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName, false, false, false, null);
        //开启发布确认
        channel.confirmSelect();
        //开始时间
        long start = System.currentTimeMillis();
        //批量发消息
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message = i + "";
            channel.basicPublish("", queueName, null, message.getBytes());
            //单个消息马上进行发布确认
            boolean flag = channel.waitForConfirms();
            if (flag) {
                System.out.println("消息发送成功");
            }
        }
        //结束时间
        long end = System.currentTimeMillis();
        System.out.println("发布" + MESSAGE_COUNT + "个单独确认消息,耗时" + (end - start) + "毫秒");
    }

    /**
     * 批量发布确认
     *
     * @throws Exception
     */
    public static void publishMessageBatch() throws Exception {
        Channel channel = RabbitmqUtils.getChannel();
        //队列的名称
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName, false, false, false, null);
        //开启发布确认
        channel.confirmSelect();
        //开始时间
        long start = System.currentTimeMillis();
        //批量确认消息大小
        int batchSize = 100;
        //批量发送消息 批量发布确认
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message = i + "";
            channel.basicPublish("", queueName, null, message.getBytes());
            //判断达到100条的时候批量确认一次
            if (i % batchSize == 0) {
                boolean flag = channel.waitForConfirms();
                if (flag) {
                    System.out.println("消息发送成功");
                }
            }
        }
        long end = System.currentTimeMillis();
        System.out.println("发布" + MESSAGE_COUNT + "个批量确认消息,耗时" + (end - start) + "毫秒");
    }

    /**
     * 异步发布确认
     * @throws Exception
     */
    public static void publishMessageAsync() throws Exception {
        Channel channel = RabbitmqUtils.getChannel();
        //队列的名称
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName, false, false, false, null);
        //开启发布确认
        channel.confirmSelect();
        //开始时间
        long start = System.currentTimeMillis();
        /**
         * 线程安全有序的hash表 适用于高并发
         * 1.轻松的将消息与序号关联
         * 2.轻松批量删除条目 只要给到序号
         * 3.支持高并发
         */
        ConcurrentSkipListMap<Long,String> concurrentSkipListMap=new ConcurrentSkipListMap<>();
        ConfirmCallback ackCallback=(deliveryTag,multiple)->{
            if(multiple){
                //删除掉已经确认的消息
                ConcurrentNavigableMap<Long, String> map = concurrentSkipListMap.headMap(deliveryTag);
                map.clear();
            }else {
                concurrentSkipListMap.remove(deliveryTag);
            }
            System.out.println("确认的消息:"+deliveryTag);
        };
        /**
         * 1.消息的标记
         * 2.是否为批量确认
         */
        ConfirmCallback nackCallback=(deliveryTag,multiple)->{
            //打印未确认的消息
            System.out.println("未确认的消息:"+deliveryTag);
        };
        //准备消息的监听器 监听哪些消息成功了 哪些失败了
        channel.addConfirmListener(ackCallback,nackCallback);

        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message = i + "";
            channel.basicPublish("", queueName, null, message.getBytes());
            //此处记录下所有要发送的消息
            concurrentSkipListMap.put(channel.getNextPublishSeqNo(),message);
        }
        long end = System.currentTimeMillis();
        System.out.println("发布" + MESSAGE_COUNT + "个异步确认消息,耗时" + (end - start) + "毫秒");
    }
}
